from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
from aiogram.utils.keyboard import InlineKeyboardBuilder


def continue_btn():
    markup = InlineKeyboardMarkup(
        inline_keyboard=[
            [
                InlineKeyboardButton(text="Продолжить", callback_data="continue"),
            ]
        ]
    )

    return markup


def buy_document_btn():
    builder = InlineKeyboardBuilder()
    builder.add(InlineKeyboardButton(text="Получить", callback_data="buy_document"))
    builder.add(InlineKeyboardButton(text="Изменить", callback_data="change_document"))
    builder.add(InlineKeyboardButton(text="Выйти", callback_data="exit"))
    builder.adjust(1, 1)

    return builder.as_markup()


def select_data_to_change():
    builder = InlineKeyboardBuilder()
    texts = [
        "Имя",
        "Фамилия",
        "Номер дома",
        "Улица",
        "Город",
        "Посткод",
    ]
    callback_datas = [
        "change_first_name",
        "change_last_name",
        "change_house_num",
        "change_street_name",
        "change_city_name",
        "change_post_code",
    ]

    for text, callback_data in zip(texts, callback_datas):
        builder.add(InlineKeyboardButton(text=text, callback_data=callback_data))

    builder.adjust(1)

    return builder.as_markup()
